传染病控制（暴力dfs,超时了）

问题 1495: [蓝桥杯][算法提高VIP]传染病控制

时间限制: 1Sec 内存限制: 128MB 提交: 37 解决: 7

题目描述

近来，一种新的传染病肆虐全球。蓬莱国也发现  了零星感染者，为防止该病在蓬莱国大范围流行，该国政府决定不惜一切代价控制传染病的蔓延。不幸的是，由于人们尚未完全认识这种传染病，难以准确判别病毒  携带者，更没有研制出疫苗以保护易感人群。于是，蓬莱国的疾病控制中心决定采取切断传播途径的方法控制疾病传播。经过  WHO（世界卫生组织）以及全球各国科研部门的努力，这种新兴传染病的传播途径和控制方法已经研究消楚，剩下的任务就是由你协助蓬莱国疾控中心制定一个有  效的控制办法。 
问题描述 
研究表明，这种传染病的传播具有两种很特殊的性质； 
第一是它的传播途径是树型的，一个人X只可能被某个特定的人Y感染，只要Y不得病，或者是XY之间的传播途径被切断，则X就不会得病。 
第二是，这种疾病的传播有周期性，在一个疾病传播周期之内，传染病将只会感染一代患者，而不会再传播给下一代。 
这些性质大大减轻了蓬莱国疾病防控的压力，并且他们已经得到了国内部分易感人群的潜在传播途径图（一棵树）。但是，麻烦还没有结束。由于蓬莱国疾控中  心人手不够，同时也缺乏强大的技术，以致他们在一个疾病传播周期内，只能设法切断一条传播途径，而没有被控制的传播途径就会引起更多的易感人群被感染（也  就是与当前已经被感染的人有传播途径相连，且连接途径没有被切断的人群）。当不可能有健康人被感染时，疾病就中止传播。所以，蓬莱国疾控中心要制定出一个  切断传播途径的顺序，以使尽量少的人被感染。你的程序要针对给定的树，找出合适的切断顺序。 

输入

输入格式的第一行是两个整数n（1≤n≤300）和p。接下来p行，每一行有两个整数i和j，表示节点i和j间有边相连（意即，第i人和第j人之间有传播途径相连，注意：可能是i到j也可能是j到i）。其中节点1是已经被感染的患者。 
对于给定的输入数据，如果不切断任何传播途径，则所有人都会感染。

输出

只有一行，输出总共被感染的人数。 

样例输入

7  6 
1  2 
1  3 
2  4 
2  5 
3  6 
7  3 

样例输出

3

提示

C语言在线学习平台微信号dotcpp

来源

算法提高

 

题目解答：

据说是用dfs暴力搜索，可是我的代码是超时了，诶。。。

先建树扫描节点，求得每一个节点的深度，depth[i]，然后dfs以深度为变量不断搜索，搜到某一层，把ill[i]==1的结点的下一代结点进行标记（传染标记），之后切断传播途径，即下一代的某一个结点置ill[i]==0,其它点继续感染下一代，注意回溯。

#include <iostream>
#include <cstring>
#include <algorithm>
#include <stack>
#include <vector>
#include <cstdio>
#include <map>
#define LL long long
#define INF 0x3f3f3f3f
using namespace std;
const int maxn=100005;
const int maxz=505;
int n,p;
vector<int>v[maxz];
int depth[maxz],ill[maxz],father[maxz];
int th,mini,sum,tem;
void buildtree(int t,int pre)
{
    depth[t]=depth[pre]+1;
    father[t]=pre;
    th=max(th,depth[t]);
    for(int i=0;i<v[t].size();i++)
    {
        if(v[t][i]!=pre)
        {
            buildtree(v[t][i],t);
        }
    }
}
void dfs(int dep)
{
    if(sum>mini)  return ;
    for(int i=1;i<=n;i++)
    {
        if(depth[i]==dep&&ill[i]==1)
        {
            for(int j=0;j<v[i].size();j++)
            {
               if(depth[v[i][j]]==dep+1)
               {
                    ill[v[i][j]]=1;
                    sum++;
                    tem++;
               }
            }
        }
    }
    for(int i=1;i<=n;i++)
    {
        if(depth[i]==dep+1&&ill[i]==1)
        {
            ill[i]=0;
            sum--;
            tem=0;
            if(th==dep+1)
                mini=min(sum,mini);
                //cout << sum << endl;
            dfs(dep+1);
            sum++;
            /*for(int u=1;u<=n;u++)
            {
                if(depth[u]==dep+1&&ill[u]==1)
                {
                    for(int j=0;j<v[u].size();j++)
                    {
                       if(depth[v[u][j]]==dep+2)
                       {
                            ill[v[u][j]]=0;
                            sum--;
                       }
                    }
                }
            }*/
            for(int u=1;u<=n;u++)
            {
                if(depth[u]==dep+2&&ill[father[u]]==1)
                {
                    ill[u]=0;
                    sum--;
                }
            }
            ill[i]=1;
        }
    }
}

int main()
{
    ios::sync_with_stdio(false);
    cin>>n>>p;
    for(int i=0;i<p;i++)
    {
        int a,b;
        cin>>a>>b;
        v[a].push_back(b);
        v[b].push_back(a);
    }
    buildtree(1,0);
    ill[1]=1;
    mini=maxz;
    sum=1;
    dfs(1);
    cout << mini << endl;
    return 0;
}